package com.san.common.utils;

import java.security.SecureRandom;

import javax.crypto.Cipher;
import javax.crypto.KeyGenerator;
import javax.crypto.SecretKey;
import javax.crypto.spec.SecretKeySpec;

import sun.misc.BASE64Decoder;
import sun.misc.BASE64Encoder;

public class AESUtils {
	/**
	 * 加密
	 * 
	 * @param keyRule 密钥生成规则
	 * @param content 加密内容
	 * @return
	 */
	public static String AESEncode(String keyRule, String content) {
		if (StringUtils.isEmpty(content)) {
			return content;
		}
		try {
			String encodeRules = Constant.SECRET_KEY + keyRule;
			// 构造密钥生成器，指定为AES算法,不区分大小写
			KeyGenerator keygen = KeyGenerator.getInstance("AES");
			SecureRandom random = SecureRandom.getInstance("SHA1PRNG","SUN");
			// 根据ecnodeRules规则初始化密钥生成器,生成一个128位的随机源,根据传入的字节数组
			random.setSeed(encodeRules.getBytes());
			keygen.init(128, random);
			// 产生原始对称密钥
			SecretKey original_key = keygen.generateKey();
			// 获得原始对称密钥的字节数组
			byte[] raw = original_key.getEncoded();
			// 根据字节数组生成AES密钥
			SecretKey key = new SecretKeySpec(raw, "AES");
			// 根据指定算法AES自成密码器
			Cipher cipher = Cipher.getInstance("AES");
			// 初始化密码器，第一个参数为加密(Encrypt_mode)或者解密解密(Decrypt_mode)操作，第二个参数为使用的KEY
			cipher.init(Cipher.ENCRYPT_MODE, key);
			// 获取加密内容的字节数组(这里要设置为utf-8)不然内容中如果有中文和英文混合中文就会解密为乱码
			byte[] byte_encode = content.getBytes("utf-8");
			// 根据密码器的初始化方式--加密：将数据加密
			byte[] byte_AES = cipher.doFinal(byte_encode);
			// 加密
			String AES_encode = new String(new BASE64Encoder().encode(byte_AES));

			return AES_encode;
		} catch (Exception e) {
			e.printStackTrace();
		}

		return null;
	}

	/**
	 * 解密
	 * 
	 * @param keyRule 密钥生成规则
	 * @param content 解密内容
	 * @return
	 */
	public static String AESDecode(String keyRule, String content) {
		try {
			String encodeRules = Constant.SECRET_KEY + keyRule;
			// 构造密钥生成器，指定为AES算法,不区分大小写
			KeyGenerator keygen = KeyGenerator.getInstance("AES");
			SecureRandom random = SecureRandom.getInstance("SHA1PRNG","SUN");
			// 根据ecnodeRules规则初始化密钥生成器,生成一个128位的随机源,根据传入的字节数组
			random.setSeed(encodeRules.getBytes());
			keygen.init(128, random);
			// 产生原始对称密钥
			SecretKey original_key = keygen.generateKey();
			// 获得原始对称密钥的字节数组
			byte[] raw = original_key.getEncoded();
			// 根据字节数组生成AES密钥
			SecretKey key = new SecretKeySpec(raw, "AES");
			// 根据指定算法AES自成密码器
			Cipher cipher = Cipher.getInstance("AES");
			// 初始化密码器，第一个参数为加密(Encrypt_mode)或者解密(Decrypt_mode)操作，第二个参数为使用的KEY
			cipher.init(Cipher.DECRYPT_MODE, key);
			// 将加密并编码后的内容解码成字节数组
			byte[] byte_content = new BASE64Decoder().decodeBuffer(content);
			// 解密
			byte[] byte_decode = cipher.doFinal(byte_content);
			String AES_decode = new String(byte_decode, "utf-8");
			return AES_decode;
		} catch (Exception e) {
			e.printStackTrace();
		}

		return null;
	}
	
	
	public static void main(String[] args) {
		String a = AESDecode("30wish@_", "/LGhjRnCbVkUkPk62xJvUQ==");
		System.out.println(a);
	}
	
}
